import streamlit as st

def display_chat():
    for message in st.session_state.chat_history:
        with st.chat_message(message["role"]):
            st.markdown(message["content"])
    
def display_user_message(user_msg):
    with st.chat_message(user_msg["role"]):
        st.markdown(user_msg["content"])